from openpyxl import load_workbook, Workbook

# todo 改成自己的
OPERATORS = [
    "牛快充（智充）", "国网分布式新能源", "智充", "河南电捷", "北京富电",
    "未来动力", "佛山市振轩南家纺有限公司", "和瑞众联科技（北京）有限公司", "浙江亿充新能源科技有限公司", "天津易电占新能源（悦鑫平台）",
    "江苏绿城信息技术有限公司", "武汉峰泊停车服务有限公司", "武汉小安极电新能源", "塑云", "天津路客畅行商贸有限公司",
    "德顺新能源科技", "深圳电王快充", "森茂新能源", "北京畅的充电", "恩逸电桩",
    "优选汽车", "东莞市东融信息咨询有限公司", "洪雁新能源科技有限公司", "速仕通", "安能绿捷",
    "诚光新能源", "广州大熊新能源科技有限公司", "杭州时代大业新能源有限公司", "河北江冀汽车租赁有限公司", "兰州海红技术股份有限公司",
    "青岛吉之行新能源汽车销售服务有限公司", "交运集团(青岛)汽车动能科技有限公司", "佛山小丫快充新能源科技有限公司", "河北泰雅环保科技有限公司", "太原易捷动力新能源有限公司",
    "山西易闪充新能源有限公司", "阳泉泽平科技有限责任公司", "广州公交集团新能源发展有限责任公司", "山西大德新能源汽车服务有限公司", "山西千丰电器设备有限公司",
    "山西森焱新能源科技有限公司", "河津市公共交通有限公司", "杭州递递叭叭", "珠海兴诺能源技术有限公司", "山西中榆电力有限公司",
    "量势新能源", "青岛鼎信通讯科技有限公司", "上海电享信息科技有限公司", "山西能之源科技有限公司", "简单充",
    "高陆通"
]
# OPERATORS = [
#     "牛快充（智充）", "国网分布式新能源", "智充", "北京富电", "未来动力", "佛山市振轩南家纺有限公司",
#     "和瑞众联科技（北京）有限公司", "恩逸电桩", "优选汽车", "浙江亿充新能源科技有限公司", "江苏绿城信息技术有限公司",
#     "天津易电占新能源（悦鑫平台）", "武汉峰泊停车服务有限公司", "武汉小安极电新能源", "塑云", "北京畅的充电",
#     "东莞市东融信息咨询有限公司", "洪雁新能源科技有限公司", "速仕通", "安能绿捷", "诚光新能源"
# ]

inFile = "C:\Download\订单列表2021091011_45_50.xlsx"
outFile = "C:\Download\挂起订单.xlsx"
# inFile = "/Users/kent/Downloads/订单列表2021072608_52_07.xlsx"
# outFile = "/Users/kent/Downloads/挂起订单.xlsx"

# 打开文件
totalXlsx = load_workbook(inFile)
totalActiveSheet = totalXlsx["Sheet1"]

# 获取文件行数
totalRowCount = totalActiveSheet.max_row

# # 新建一个空白文件
# newXlsx = Workbook()
# newActiveSheet = newXlsx.active

# # 新建一个Sheet
# newActiveSheet = totalXlsx.create_sheet("Sheet2")

# 遍历G列
colG = totalActiveSheet["G"]
for gCell in colG:
    # 过滤出属于自己的运营商
    if gCell.value in OPERATORS:
        # 找出单元格所在的行
        gCellRow = totalActiveSheet[gCell.row]
        # 将行追加到文件末尾
        totalActiveSheet.append(gCellRow)

# 删除原有行
totalActiveSheet.delete_rows(2, totalRowCount-1)

# 保存表
totalXlsx.save(outFile)
